package xyz.sekai52799.answer.a601_a800;

import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

public class Question648 {
//    在英语中，我们有一个叫做词根(root) 的概念，可以词根后面添加其他一些词组成另一个较长的单词——我们称这个词为继承词(successor)。
//    例如，词根an，跟随着单词other(其他)，可以形成新的单词another(另一个)。
//
//    现在，给定一个由许多词根组成的词典 dictionary 和一个用空格分隔单词形成的句子 sentence。你需要将句子中的所有继承词用词根替换掉。
//    如果继承词有许多可以形成它的词根，则用最短的词根替换它。
//
//    你需要输出替换之后的句子。
//
//
//
//    示例 1：
//
//    输入：dictionary = ["cat","bat","rat"], sentence = "the cattle was rattled by the battery"
//    输出："the cat was rat by the bat"
//    示例 2：
//
//    输入：dictionary = ["a","b","c"], sentence = "aadsfasf absbs bbab cadsfafs"
//    输出："a a b c"
//
//
//    提示：
//
//            1 <= dictionary.length<= 1000
//            1 <= dictionary[i].length <= 100
//    dictionary[i]仅由小写字母组成。
//            1 <= sentence.length <= 10^6
//    sentence仅由小写字母和空格组成。
//    sentence 中单词的总量在范围 [1, 1000] 内。
//    sentence 中每个单词的长度在范围 [1, 1000] 内。
//    sentence 中单词之间由一个空格隔开。
//    sentence没有前导或尾随空格。
//
//    来源：力扣（LeetCode）
//    链接：https://leetcode.cn/problems/replace-words
//    著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

    // 我的思路是，先把dictionary按从短到长的顺序排序，然后遍历替换并组装到StringBuilder里面
    public String replaceWords(List<String> dictionary, String sentence) {
        dictionary.sort(new CompareByLength());
        StringBuilder stringBuilder = new StringBuilder();
        String[] strings = sentence.split(" ");
        for (int i = 0;i < strings.length;i++) {
            for (String dict : dictionary){
                if (strings[i].startsWith(dict)){
                    strings[i] = dict;
                    break;
                }
            }
            stringBuilder.append(strings[i]);
            if (i < strings.length - 1){
                stringBuilder.append(" ");
            }
        }

        return stringBuilder.toString();
    }

    private static class CompareByLength implements Comparator<String> {
        @Override
        public int compare(String s1, String s2) {
            int l1 = s1.length();
            int l2 = s2.length();
            return l1-l2;
        }
    }
}
